Implementation Notes

Continuous Distribution Implementation

The continuous wage offer distribution uses:

• Lognormal wages: w(s) = exp(μ + σs) where s ~ N(0,1)
• Monte Carlo integration: Averages over 1000 wage draws to evaluate integrals
• JAX JIT compilation: Significant speedup for iterative computations

Volatility Analysis

The mean-preserving spread is achieved by adjusting μ when varying σ:

• For lognormal distribution, mean = exp(μ + σ²/2)
• To maintain constant mean m: μ = ln(m) - σ²/2
• Results show reservation wage increasing from ~18 to ~21 as σ goes from 0.1 to 1.0

Code Quality

• All code follows existing style conventions in the lecture
• Uses JAX's functional programming paradigm with jax.lax.while_loop
• Vectorized computations with jax.vmap for efficiency
• Clear documentation in docstrings and comments

Testing

Tested via jupytext --to py mccall_model.md and execution:

• ✅ All cells execute without errors
• ✅ Contour plots generate correctly
• ✅ Volatility visualization shows expected increasing trend
• ✅ Exercise solution runs successfully with 100,000 Monte Carlo replications

The file is ready for review and merge.

📖 Netlify Preview Ready!

Preview URL: https://pr-709--sunny-cactus-210e3e.netlify.app (2ea4ce0)

📚 Changed Lecture Pages: mccall_fitted_vfi, mccall_model

Added new section "Lifetime Value and Volatility" that demonstrates how expected lifetime value increases with volatility under the optimal policy.

Implementation

The section includes:

• Parallel JAX simulation code using jax.vmap to run 10,000 replications efficiently
• simulate_lifetime_value(): JIT-compiled function that simulates job search with jax.lax.while_loop
• compute_mean_lifetime_value(): Vectorized computation across all simulation replications
• Plot showing expected lifetime value increasing with σ

Key Results

For each volatility level:

1. Computes the optimal reservation wage
2. Simulates workers following the optimal policy (accept if wage ≥ reservation wage)
3. Calculates expected discounted lifetime income over 10,000 replications
4. Demonstrates that higher volatility → higher lifetime value (option value of search)

The code runs without warnings and complements the existing volatility section that shows reservation wages increase with σ.

🤖 Generated with Claude Code

📖 Netlify Preview Ready!

Preview URL: https://pr-709--sunny-cactus-210e3e.netlify.app (905151e)

📚 Changed Lecture Pages: mccall_fitted_vfi, mccall_model

Simplified the lifetime value simulation for better parallelization:

Changes

Replaced the while_loop based simulation with a fixed 100-period approach:

Before: Used jax.lax.while_loop to search until acceptance, then computed analytical present value
After: Draw all 100 wage offers upfront and process vectorially

Key Improvements

1. Cleaner logic: No nested loop functions, just straightforward array operations
2. Better parallelization: All simulation paths have the same length
3. Simpler code:
   • jnp.cumsum(accept) > 0 tracks when employment starts
   • jnp.argmax(accept) finds first accepted wage
   • Vectorized computation of earnings and discounted sum

Code Structure

# Draw 100 wage offers upfront
wage_offers = jnp.exp(μ + σ * s_vals)

# Track employment from first acceptance onward
employed = jnp.cumsum(wage_offers >= w_bar) > 0

# Compute earnings and discounted sum
earnings = jnp.where(employed, accepted_wage, c)
lifetime_value = jnp.sum(discount_factors * earnings)

This approach is cleaner and more efficient for parallel JAX execution, while 100 periods provides sufficient accuracy given β=0.99.

🤖 Generated with Claude Code

📖 Netlify Preview Ready!

Preview URL: https://pr-709--sunny-cactus-210e3e.netlify.app (326b343)

📚 Changed Lecture Pages: mccall_fitted_vfi, mccall_model

📖 Netlify Preview Ready!

Preview URL: https://pr-709--sunny-cactus-210e3e.netlify.app (e609076)

📚 Changed Lecture Pages: mccall_fitted_vfi, mccall_model